export const transformerTreeData = [
  {
    label: 'Transformer 模型',
    children: [
      {
        label: '编码器',
        children: [
          {
            label: '输入(相加)',
            children: [
              {
                label: 'Token Embedding',
              },
              {
                label: 'Positional Encoding',
              },
            ],
          },
          {
            label: '编码块-Layer-0',
            children: [
              {
                label: '注意力子块',
                children: [
                  {
                    label: 'MultiHead Attention'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
              {
                label: '前向子块',
                children: [
                  {
                    label: 'Feed Forward'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
            ],
          },
          {
            label: '编码块-Layer-1',
            children: [
              {
                label: '注意力子块',
                children: [
                  {
                    label: 'MultiHead Attention'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
              {
                label: '前向子块',
                children: [
                  {
                    label: 'Feed Forward'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
            ],
          },
          {
            label: '编码块-Layer-2',
            children: [
              {
                label: '注意力子块',
                children: [
                  {
                    label: 'MultiHead Attention'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
              {
                label: '前向子块',
                children: [
                  {
                    label: 'Feed Forward'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
            ],
          },
          {
            label: '编码块-Layer-3',
            children: [
              {
                label: '注意力子块',
                children: [
                  {
                    label: 'MultiHead Attention'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
              {
                label: '前向子块',
                children: [
                  {
                    label: 'Feed Forward'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
            ],
          },
          {
            label: '编码块-Layer-4',
            children: [
              {
                label: '注意力子块',
                children: [
                  {
                    label: 'MultiHead Attention'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
              {
                label: '前向子块',
                children: [
                  {
                    label: 'Feed Forward'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
            ],
          },
          {
            label: '编码块-Layer-5',
            children: [
              {
                label: '注意力子块',
                children: [
                  {
                    label: 'MultiHead Attention'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
              {
                label: '前向子块',
                children: [
                  {
                    label: 'Feed Forward'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
            ],
          },
        ],
      },
      {
        label: '解码器',
        children: [
          {
            label: '输入(相加)',
            children: [
              {
                label: 'Token Embedding(训练用，推理不用)',
              },
              {
                label: 'Positional Encoding',
              },
            ],
          },
          {
            label: '解码块-Layer-0',
            children: [
              {
                label: '注意力子块',
                children: [
                  {
                    label: 'MultiHead Attention'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
              {
                label: '前向子块',
                children: [
                  {
                    label: 'Feed Forward'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
            ],
          },
          {
            label: '解码块-Layer-1',
            children: [
              {
                label: '注意力子块',
                children: [
                  {
                    label: 'MultiHead Attention'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
              {
                label: '前向子块',
                children: [
                  {
                    label: 'Feed Forward'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
            ],
          },
          {
            label: '解码块-Layer-2',
            children: [
              {
                label: '注意力子块',
                children: [
                  {
                    label: 'MultiHead Attention'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
              {
                label: '前向子块',
                children: [
                  {
                    label: 'Feed Forward'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
            ],
          },
          {
            label: '解码块-Layer-3',
            children: [
              {
                label: '注意力子块',
                children: [
                  {
                    label: 'MultiHead Attention'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
              {
                label: '前向子块',
                children: [
                  {
                    label: 'Feed Forward'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
            ],
          },
          {
            label: '解码块-Layer-4',
            children: [
              {
                label: '注意力子块',
                children: [
                  {
                    label: 'MultiHead Attention'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
              {
                label: '前向子块',
                children: [
                  {
                    label: 'Feed Forward'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
            ],
          },
          {
            label: '解码块-Layer-5',
            children: [
              {
                label: '注意力子块',
                children: [
                  {
                    label: 'MultiHead Attention'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
              {
                label: '前向子块',
                children: [
                  {
                    label: 'Feed Forward'
                  },
                  {
                    label: 'Layer Norm'
                  }
                ]
              },
            ],
          },
          {
            label: '全连接层',
          },
          {
            label: 'Softmax(输出分类概率)',
          },
        ],
      },
    ]
  }
]

export const transformerTreeProps = {
  children: 'children',
  label: 'label',
}
